package module

import (
	"bufio"
	"fmt"
	"os"
	"strings"
)

// InputIp 从用户输入中添加记录
func (h *Hosts) InputIp() string {
	reader := bufio.NewReader(os.Stdin)
	for {
		fmt.Println("请输入IP地址: ")
		ip, _ := reader.ReadString('\n')
		ip = strings.TrimSpace(ip)
		if isValidIP(ip) {
			return ip
		}
		fmt.Println("IP地址格式不正确，请重新输入")
	}
}

// InputDomain 从用户输入中添加记录
func (h *Hosts) InputDomain() string {
	reader := bufio.NewReader(os.Stdin)
	for {
		fmt.Println("请输入域名: ")
		domain, _ := reader.ReadString('\n')
		domain = strings.TrimSpace(domain)
		if isValidHostname(domain) {
			return domain
		}
		fmt.Println("域名格式不正确，请重新输入")
	}
}
